# -*- coding: utf-8 -*-
# @Author  : Zhang.Jingyi

from datasets.dataloader import BaseDataset


def pedx_data_generator(options, is_train=True, is_use_image=False, is_aug=False, is_visual=False):
    data = BaseDataset(options=options, dataset='pedx', is_train=is_train,
                       is_use_image=is_use_image, is_aug=is_aug, is_visual=is_visual)
    return data


def pedx_sparse_generator(options, is_train=True, is_use_image=False, is_aug=False, is_visual=False):
    data = BaseDataset(options=options, dataset='pedx', is_train=is_train,
                       is_use_image=is_use_image, is_aug=is_aug, is_visual=is_visual)
    point_xyz = data['human_points']
